import numpy as np 

lis=[]
file=open("log.txt")

for line in file.readlines():
    if line==' \n':
        lis.append(line)#如果是空行，则代表是帧分割，所以直接continue就好
        continue
    curline=line.strip().split(" ")
    if len(curline)==7:
        curline[0]=curline[0]+curline[1]#有时候label的标签会分开，例如stopsign会变成stop  sign，所以要检测是否长度都为6，不是的话就合并一下
        curline.remove(curline[1])

    floatline=float(curline[1])#将阈值转换为float
    if(floatline<0.5):#如果阈值小就不要，否则就插入lis里面
        continue
    else:
        lis.append(curline)
lis.remove(lis[0])#为了配准要删除前两个
lis.remove(lis[1])
#files=open("log_threhold.txt","w")






ll=[]
lll=[]
for f in lis:#具体做法就是没有换行的时候把每一行都输入ll，碰到换行了就把ll输入lll，之后ll归零
    if(f==' \n'):
        lll.append(ll)
        ll=[]
    else:
        ll.append(f)

w=np.array(lll)
np.save("yolo.npy",w)#存为npy
# for fp in lis:
#     files.write(str(fp))
# files.close()